torchrun --nproc_per_node=8 --nnodes 1 --master_port 16666 main_pretrain.py --batch_size 256 --selectivemae --dataset opticalrs-4m --dataset_path '/home/wangfengxiang/data/imagenet/imagenet/train' --model mae_vit_base_patch16 --output_dir output --norm_pix_loss --blr 1.5e-4 --weight_decay 0.05  --num_workers 12  --decoder_depth 8 --mask_ratio 0.85 --kept_mask_ratio 0.25 --epochs 800 --warmup_epochs 10
